import time as _time
import torch as _torch
from src.networks import SimpleConv as _net

_name = 'SimpleConvFiltered8641'

# MODEL PARA
USE_NET = _net
###############

# TRAIN SCHEME
USE_SAMPLE_FILTERED = True
BATCH_SIZE = 32
VAL_BATCH_SIZE = 32

LEARNING_RATE = 0.0001
NUM_EPOCH = 100 
LEARNING_RATE_DECAY = 0.98

TRAIN_DATASET_MODE = 'AFFINE'
VAL_DATASET_MODE = 'CENTER'
##############

# DO NOT MODIFY
_time = _time.strftime('%m-%d %H:%M:%S', _time.localtime())
TRAIN_DIR = 'runs/%s/' % _name + _time
VAL_DIR = 'runs/%s/' % _name + _time
MODEL_NAME = '%s.pkl' % _name
device = _torch.device('cuda:0' if _torch.cuda.is_available() else 'cpu')
############

label_name2id = {
    'normal': 0,
    'yawn': 1,
    'fatigue': 2,
}

label_id2name = {
    0: 'normal',
    1: 'yawn',
    2: 'fatigue'
}



